# 调用pytorch
import torch

# torch.set_default_tensor_type(t)
# 示例
# 将默认张量类型设置为cpu和32位
torch.set_default_tensor_type(torch.FloatTensor)
# 将默认张量类型设置为cpu和64位
torch.set_default_tensor_type(torch.DoubleTensor)
# 将默认张量类型设置为gpu和32位
torch.set_default_tensor_type(torch.cuda.FloatTensor)
# 将默认张量类型设置为gpu和64位
torch.set_default_tensor_type(torch.cuda.DoubleTensor)
# 将默认张量类型设置为gpu和16位
torch.set_default_tensor_type(torch.cuda.HalfTensor)
# 将默认张量类型设置为gpu和8位
torch.set_default_tensor_type(torch.cuda.BFloat16Tensor)

